import taichi as ti

ti.init()

n,m=3,3
u = ti.field(ti.f32,shape=(n,m))
v = ti.field(ti.f32,shape=(n,m))
p = ti.field(ti.f32,shape=(n,m))